package response

type UsageResp struct {
	Prompt           string `json:"prompt,omitempty"`  // 用户提示词
	Content          string `json:"content,omitempty"` // 模型响应内容
	PromptTokens     int    `json:"prompt_tokens"`     // 用户输入的token数量
	CompletionTokens int    `json:"completion_tokens"` // 模型输出的token数量
	TotalTokens      int    `json:"total_tokens"`      // 总token数量
}

type ChoicesResp struct {
	Index   int `json:"index"`
	Message struct {
		Role    string `json:"role"`
		Content string `json:"content"`
	} `json:"message"`
	Logprobs     interface{} `json:"logprobs"`
	FinishReason string      `json:"finish_reason"`
}

type OpenAIResp struct {
	Id                string        `json:"id"`
	Object            string        `json:"object"`
	Created           int           `json:"created"`
	Model             string        `json:"model"`
	SystemFingerprint string        `json:"system_fingerprint"`
	Choices           []ChoicesResp `json:"choices"`
	Usage             UsageResp     `json:"usage"`
}
